package com.hy;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created With IntelliJ IDEA.
 * Descriptions:二叉搜索树中的众数
 * 给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。
 * 如果树中有不止一个众数，可以按 任意顺序 返回。
 *
 * 提示：
 * 树中节点的数目在范围 [1, 104] 内
 * -105 <= Node.val <= 105
 *
 * User:Mr.Du
 * Date:2023/9/17
 * Time:11:27
 */
public class FindMode {

    int pre = -1000000;
    int count = 1;
    int max = 0;
    List<Integer> list = new ArrayList<>();
    public int[] findMode(TreeNode root) {
        dfs(root);
        int[] res = new int[list.size()];
        int index = 0;
        for(int i : list){
            res[index++] = i;
        }
        return res;
    }

    /**
     * 利用二叉搜索树的性质
     * @param root
     */
    public void dfs(TreeNode root){
        if(root == null) return;
        dfs(root.left);
        if(pre == root.val){
            count++;
        }else{
            count = 1;
        }
        pre = root.val;
        if(count == max) list.add(root.val);
        if(count > max){
            max = count;
            list.clear();
            list.add(root.val);
        }
        dfs(root.right);
    }


    Map<Integer, Integer> mp = new HashMap<>();

    public int[] findModeOrdinary(TreeNode root) {
        travel(root);
        for(Map.Entry<Integer, Integer> es : mp.entrySet()){
            int key = es.getKey();
            int value = es.getValue();
            if(value == max){
                list.add(key);
            }else if(value < max){
                list.clear();
                max = value;
                list.add(key);
            }
        }
        int[] res = new int[list.size()];
        int index = 0;
        for(int i : list){
            res[index++] = i;
        }
        return res;
    }

    /**
     * 普通二叉树用map进行统计
     * @param root
     */
    public void travel(TreeNode root){
        if(root == null) return;
        mp.put(root.val, mp.getOrDefault(root.val, 0) + 1);
        travel(root.left);
        travel(root.right);
    }

}
